#include "BasicBlock.h"
#include "Constant.h"
#include "Function.h"
#include "IRBuilder.h"
#include "Module.h"
#include "Type.h"

#include <iostream>
#include <memory>

#ifdef DEBUG  // 用于调试信息,大家可以在编译过程中通过" -DDEBUG"来开启这一选项
#define DEBUG_OUTPUT std::cout << __LINE__ << std::endl;  // 输出行号的简单示例
#else
#define DEBUG_OUTPUT
#endif

#define CONST_INT(num) \
 ConstantInt::get(num, module)

#define CONST_FP(num) \
 ConstantFP::get(num, module) // 得到常数值的表示,方便后面多次用到

int main() {
    auto module = new Module("Cminus code");            // module name是什么无关紧要
    auto builder = new IRBuilder(nullptr, module);      // 创建IRBuilder
    Type* Int32Type = Type::get_int32_type(module);

    // main函数
    auto mainFun = Function::create(FunctionType::get(Int32Type, {}),   /* 创建 main 函数 */
                                    "main", module);
    auto bb = BasicBlock::create(module, "main", mainFun);              /* 创建基本块，命名为main */
    builder->set_insert_point(bb);                              /* 将基本块加入到builder中 */
    // 设置默认返回
    auto retAlloca = builder->create_alloca(Int32Type);         /* 创建返回默认量 */
    builder->create_store(CONST_INT(0), retAlloca);             /* 给默认量赋0，表示默认ret 0 */
    // 具体执行
    Type* FloatType = Type::get_float_type(module);             /* 获取单个float类型的指针 */
    auto aAlloca = builder->create_alloca(FloatType);           /* 根据float类型的指针，申请一个float变量空间 */
    builder->create_store(CONST_FP(5.555), aAlloca);            /* 将值5.555存入该变量空间 */

    auto a = builder->create_load(aAlloca);                     /* 取出该变量空间内的值，即a的值 */
    auto fcmp = builder->create_fcmp_gt(a, CONST_FP(1.00));     /* 将其和1.00进行比较，返回结果存到fcmp中 */

    auto trueBB = BasicBlock::create(module, "trueBB", mainFun);/* 符合if条件的分支 */
    auto falseBB = BasicBlock::create(module, "falseBB", mainFun);  /* 不符合if条件的分支 */
    builder->create_cond_br(fcmp, trueBB, falseBB);             /* 根据fcmp创建跳转语句 */

    builder->set_insert_point(trueBB);      // if true; 分支的开始需要SetInsertPoint设置
    builder->create_ret(CONST_INT(233));    /* 创建返回，将值233返回 */

    builder->set_insert_point(falseBB);     // if false; 分支的开始需要SetInsertPoint设置
    builder->create_ret(CONST_INT(0));      /* 创建返回，将值0返回 */

    std::cout << module->print();
    delete module;

    return 0;
}
